# jishe/permissions.py
from rest_framework import permissions

class IsOwnerOrSuperUser(permissions.BasePermission):
    """
    自定义权限只允许对象的创建者才能编辑它。
    其他人只读
    """

    def has_object_permission(self, request, view, obj):
        # 写入权限只允许给 article 的作者。
        # if request.user.is_superuser == True:
        #     return True

        return obj.user == request.user



class ProfileUnique(permissions.BasePermission):
    """
    一次只能提交一个档案
    """
    def has_permission(self, request, view):
        # 写入权限只允许给 article 的作者。
        return not request.user.commit_profile.count()